from random import choice


class RandomWalk:
    #生成一个随机游走数据的类
    def __init__(self,num_points=5000):
        self.num_points=num_points
        #表示随机游走的初始点是（0,0）
        self.x_values=[0]
        self.y_values=[0]
    def fill_walk(self):
        while len(self.x_values)<self.num_points:
            x_direction=choice([1,-1])
            x_distance=choice([0,1,2,3,4])
            x_step=x_direction*x_distance

            y_direction = choice([1, -1])
            y_distance = choice([0, 1, 2, 3, 4])
            y_step = y_direction * y_distance

            if x_step==0 and y_step==0:
                continue
            #计算下一个点的X值和y值
            y=self.y_values[-1]+y_step
            x=self.x_values[-1]+x_step

            self.x_values.append(x)
            self.y_values.append(y)

